'''
被测试系统的接口封装
获取验证码：
登录：
'''
#导包

#定义类
class loginAPI():
    #存放调用的变量（使用__init__）
    def __init__(self):
        self.url_verify = "http://localhost/index.php?m=Home&c=User&a=verify"
        self.url_login = "http://localhost/index.php?m=Home&c=User&a=do_login"
    #获取验证码（session为scripts目录文件中调用的session对象）
    def get_verify(self, session):
        #使用session对象获取验证码
        return session.get(self.url_verify)

    #进行登录（session, username, password, verify_code对应scripts目录文件调用的session对象和username, password, verify_code参数）
    def login(self, session, username, password, verify_code):
        #使用scripts目录文件的参数
        login_data = {
            'username':username,
            'password':password,
            'verify_code':verify_code
        }
        #使用session对象惊醒登录并获取登录结果
        return session.post(url=self.url_login, data=login_data)


